package com.artisan.controller;


import jakarta.annotation.PostConstruct;
import lombok.extern.slf4j.Slf4j;
import org.springframework.ai.chat.client.ChatClient;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

import java.util.Map;

@Slf4j
@RestController
public class MultiModelsController {

    /**
     * 自动注入的聊天客户端映射表
     * 用于存储和管理不同类型的聊天客户端实例
     * key为客户端类型标识，value为对应的聊天客户端实例
     */
    @Autowired
    private Map<String, ChatClient> chatClientMap;

    /**
     * 初始化方法，用于初始化聊天客户端映射表
     */
    @PostConstruct
    public void init(){
        // 打印模型列表
        chatClientMap.forEach((key, value) -> {
            log.info("model: {} : {}", key, value.getClass().getName());
        }) ;
    }

    /**
     * 处理聊天请求并生成回复内容
     * @param message 用户输入的聊天消息
     * @param model 指定使用的聊天模型名称
     * @return 聊天模型生成的回复内容
     */
    @GetMapping("/chat")
    String generation(@RequestParam String message,
                      @RequestParam String model) {
        // 根据指定模型获取对应的聊天客户端
        ChatClient chatClient = chatClientMap.get(model);
        // 构造聊天提示并调用模型生成回复内容
        String content = chatClient.prompt().user(message).call().content();
        log.info("model: {} : {}", model, content);
        return content;
    }

}